// Copyright 2021 ETH Zurich and University of Bologna.
// Solderpad Hardware License, Version 0.51, see LICENSE for details.
// SPDX-License-Identifier: SHL-0.51
//
// Author: Matheus Cavalcante <matheusd@iis.ee.ethz.ch>
//         Basile Bougenot <bbougenot@student.ethz.ch>

#include "vector_macros.h"

void TEST_CASE1() {
  VSET(1, e8, m1);
  VLOAD_8(v1, 0b10001001);
  VSET(8, e8, m1);
  asm volatile("viota.m v2, v1");
  VCMP_U8(1, v2, 0, 1, 1, 1, 2, 2, 2, 2);

  VSET(2, e8, m1);
  VLOAD_8(v1, 0b01100010, 0b01001100);
  VSET(16, e8, m1);
  asm volatile("viota.m v2, v1");
  VCMP_U8(2, v2, 0, 0, 1, 1, 1, 1, 2, 3, 3, 3, 3, 4, 5, 5, 5, 6);

  VSET(1, e8, m8);
  VLOAD_8(v0, 0b00000001);
  VSET(4, e16, m8);
  asm volatile("viota.m v8, v0");
  VCMP_U16(3, v8, 0, 1, 1, 1);

  VSET(64, e8, m4);
  VLOAD_8(
      v0, 0b10101101, 0b10000000, 0b00000110, 0b10011100, 0b10010101,
      0b01001100, 0b01101010, 0b11000100, 0b00011110, 0b10111010, 0b00100110,
      0b11001010, 0b01101101, 0b11001010, 0b01000101, 0b00010110, 0b00001000,
      0b10111000, 0b11011100, 0b11100000, 0b00110101, 0b10011110, 0b01001111,
      0b01011101, 0b00001010, 0b01111000, 0b11010100, 0b01011101, 0b11000101,
      0b10010010, 0b01011100, 0b11010101, 0b00100010, 0b10000100, 0b11001011,
      0b01001101, 0b01010000, 0b10110011, 0b00011000, 0b10000101, 0b01110101,
      0b00001111, 0b10111100, 0b00010101, 0b10011101, 0b11011001, 0b11010101,
      0b00100001, 0b01101110, 0b10000001, 0b01100100, 0b00010001, 0b00010100,
      0b00101011, 0b11111000, 0b10010000, 0b01010000, 0b01001111, 0b00000011,
      0b10100100, 0b10001010, 0b01110011, 0b10100010, 0b01111110);
  VSET(512, e8, m4);
  asm volatile("viota.m v8, v0");
  VCMP_U8(
      4, v8, 0, 1, 1, 2, 3, 3, 4, 4, 5, 5, 5, 5, 5, 5, 5, 5, 6, 6, 7, 8, 8, 8,
      8, 8, 8, 8, 8, 9, 10, 11, 11, 11, 12, 13, 13, 14, 14, 15, 15, 15, 16, 16,
      16, 17, 18, 18, 18, 19, 19, 19, 20, 20, 21, 21, 22, 23, 23, 23, 23, 24,
      24, 24, 24, 25, 26, 26, 27, 28, 29, 30, 30, 30, 30, 30, 31, 31, 32, 33,
      34, 34, 35, 35, 36, 37, 37, 37, 38, 38, 38, 38, 39, 39, 40, 40, 40, 41,
      42, 43, 43, 44, 45, 45, 46, 47, 47, 47, 48, 48, 49, 49, 49, 50, 51, 52,
      52, 53, 53, 53, 53, 54, 54, 54, 55, 56, 56, 57, 57, 57, 57, 57, 57, 57,
      58, 58, 58, 58, 58, 58, 58, 58, 59, 60, 61, 61, 62, 62, 62, 63, 64, 65,
      65, 66, 67, 67, 67, 67, 67, 67, 68, 69, 70, 71, 71, 72, 72, 73, 74, 74,
      74, 74, 75, 76, 77, 78, 78, 78, 79, 80, 81, 82, 83, 83, 83, 84, 84, 85,
      85, 86, 87, 88, 88, 89, 89, 89, 90, 90, 91, 91, 91, 91, 91, 91, 91, 91,
      92, 93, 94, 95, 95, 95, 95, 96, 96, 97, 97, 98, 99, 100, 100, 101, 102,
      103, 103, 104, 104, 105, 105, 106, 106, 106, 106, 107, 108, 108, 109, 109,
      109, 110, 110, 110, 111, 111, 111, 112, 113, 114, 114, 115, 115, 116, 116,
      117, 117, 118, 118, 119, 120, 120, 121, 121, 121, 121, 122, 122, 122, 122,
      122, 123, 123, 123, 123, 123, 124, 125, 126, 126, 127, 127, 127, 128, 129,
      130, 130, 131, 132, 132, 132, 133, 133, 133, 133, 133, 133, 134, 134, 135,
      135, 136, 137, 137, 137, 138, 139, 139, 140, 140, 140, 140, 141, 142, 142,
      142, 142, 143, 143, 144, 144, 144, 144, 144, 145, 146, 146, 147, 147, 148,
      149, 150, 150, 151, 152, 153, 154, 154, 154, 154, 154, 154, 154, 155, 156,
      157, 158, 158, 159, 160, 160, 161, 161, 162, 162, 162, 162, 163, 163, 164,
      165, 166, 166, 166, 167, 168, 168, 168, 169, 170, 170, 171, 172, 173, 173,
      174, 174, 175, 175, 176, 177, 178, 178, 178, 178, 178, 179, 179, 179, 179,
      180, 181, 182, 182, 183, 184, 184, 185, 185, 185, 185, 185, 185, 185, 186,
      186, 186, 187, 187, 187, 188, 189, 189, 190, 190, 190, 190, 191, 191, 191,
      191, 191, 191, 192, 192, 193, 193, 193, 193, 194, 195, 195, 196, 196, 197,
      197, 197, 197, 197, 197, 198, 199, 200, 201, 202, 202, 202, 202, 202, 203,
      203, 203, 204, 204, 204, 204, 204, 205, 205, 206, 206, 207, 208, 209, 210,
      210, 210, 211, 211, 212, 213, 213, 213, 213, 213, 213, 213, 213, 213, 214,
      214, 214, 215, 215, 216, 216, 217, 217, 218, 218, 218, 218, 219, 220, 221,
      221, 221, 222, 223, 224, 224, 224, 225, 225, 225, 225, 226, 226, 227, 227,
      228, 229, 230, 231, 232, 233);

  VSET(128, e8, m8);
  VLOAD_8(
      v0, 0b00000001, 0b00000011, 0b00111110, 0b00000100, 0b01101000,
      0b10111011, 0b11010110, 0b10111111, 0b00110011, 0b00011100, 0b11010100,
      0b00011010, 0b10100001, 0b10110100, 0b10010111, 0b01010100, 0b00011010,
      0b01101011, 0b00101010, 0b11111111, 0b10000100, 0b11100110, 0b00100001,
      0b01101000, 0b10110100, 0b01100010, 0b11100001, 0b10011100, 0b00110111,
      0b01010011, 0b01010111, 0b10010001, 0b11001000, 0b01001011, 0b01000000,
      0b10001111, 0b00001111, 0b01110100, 0b10101100, 0b00010101, 0b00110100,
      0b10010010, 0b00001101, 0b11110011, 0b10101101, 0b10000100, 0b01111000,
      0b11010101, 0b10110110, 0b00110110, 0b01010001, 0b01001000, 0b11100011,
      0b01001110, 0b11101101, 0b01111000, 0b10111101, 0b00111011, 0b10111001,
      0b11000110, 0b00000011, 0b00001110, 0b00001111, 0b00000010, 0b01010110,
      0b00000010, 0b11011011, 0b01010100, 0b10110110, 0b10100011, 0b10100101,
      0b11110101, 0b00000110, 0b10011111, 0b01000110, 0b00100000, 0b00100011,
      0b11110100, 0b10111101, 0b10000010, 0b11110011, 0b00111111, 0b11000010,
      0b00011001, 0b10000010, 0b00110011, 0b11000110, 0b11001100, 0b10011100,
      0b11001011, 0b10101101, 0b11011110, 0b11010110, 0b11010110, 0b00100100,
      0b01111010, 0b00111001, 0b10111000, 0b01101000, 0b00001001, 0b10010100,
      0b11111101, 0b00001101, 0b10100111, 0b11000110, 0b01100111, 0b01010111,
      0b10011001, 0b01100111, 0b00001011, 0b01001011, 0b10001101, 0b11110110,
      0b10001001, 0b10010101, 0b10010010, 0b10100100, 0b01010110, 0b10110110,
      0b10111001, 0b01010000, 0b01010001, 0b01001100, 0b11001101, 0b10111100,
      0b11110010, 0b00001000, 0b10111000);
  VSET(1024, e16, m8);
  asm volatile("viota.m v8, v0");
  VCMP_U16(
      5, v8, 0, 1, 1, 1, 1, 1, 1, 1, 1, 2, 3, 3, 3, 3, 3, 3, 3, 3, 4, 5, 6, 7,
      8, 8, 8, 8, 8, 9, 9, 9, 9, 9, 9, 9, 9, 9, 10, 10, 11, 12, 12, 13, 14, 14,
      15, 16, 17, 17, 18, 18, 19, 20, 20, 21, 21, 22, 23, 24, 25, 26, 27, 28,
      29, 29, 30, 31, 32, 32, 32, 33, 34, 34, 34, 34, 34, 35, 36, 37, 37, 37,
      37, 37, 37, 38, 38, 39, 39, 40, 41, 41, 42, 42, 43, 44, 44, 44, 44, 45,
      45, 45, 45, 45, 46, 46, 47, 47, 47, 48, 48, 49, 50, 50, 51, 52, 53, 54,
      54, 55, 55, 55, 56, 56, 56, 57, 57, 58, 58, 59, 59, 59, 60, 60, 61, 62,
      62, 62, 62, 63, 64, 64, 65, 65, 66, 67, 67, 67, 68, 68, 69, 69, 70, 70,
      70, 71, 72, 73, 74, 75, 76, 77, 78, 78, 78, 79, 79, 79, 79, 79, 80, 80,
      81, 82, 82, 82, 83, 84, 85, 86, 86, 86, 86, 86, 87, 87, 87, 87, 87, 87,
      88, 88, 89, 90, 90, 90, 90, 91, 91, 92, 93, 93, 94, 94, 95, 95, 95, 95,
      96, 97, 97, 98, 98, 98, 98, 98, 99, 100, 101, 101, 101, 102, 103, 104,
      104, 104, 105, 106, 107, 108, 108, 109, 110, 110, 110, 111, 112, 112, 112,
      113, 113, 114, 114, 115, 116, 117, 117, 118, 118, 119, 119, 120, 120, 120,
      120, 121, 121, 121, 122, 122, 122, 122, 123, 123, 123, 124, 125, 126, 127,
      127, 128, 128, 128, 129, 129, 129, 129, 129, 129, 129, 129, 130, 130, 131,
      132, 133, 134, 134, 134, 134, 135, 136, 137, 138, 139, 139, 139, 139, 139,
      139, 139, 140, 140, 141, 142, 143, 143, 143, 143, 144, 145, 145, 146, 146,
      147, 148, 148, 149, 149, 150, 150, 150, 150, 150, 150, 151, 151, 152, 153,
      153, 153, 153, 154, 154, 154, 155, 155, 155, 156, 157, 157, 158, 159, 159,
      159, 159, 159, 160, 161, 161, 161, 162, 163, 164, 165, 166, 166, 167, 168,
      168, 169, 169, 170, 170, 170, 171, 171, 171, 171, 171, 172, 172, 172, 172,
      173, 174, 175, 176, 176, 177, 177, 178, 178, 179, 179, 180, 181, 181, 182,
      183, 183, 184, 185, 185, 186, 186, 187, 188, 188, 189, 190, 190, 190, 191,
      191, 191, 191, 192, 192, 193, 193, 193, 193, 193, 194, 194, 194, 195, 195,
      196, 197, 197, 197, 197, 198, 199, 200, 200, 201, 202, 203, 203, 203, 204,
      204, 205, 205, 206, 207, 207, 208, 209, 210, 210, 210, 210, 211, 212, 213,
      214, 214, 215, 215, 216, 217, 218, 219, 219, 220, 221, 222, 222, 223, 224,
      225, 225, 225, 226, 226, 226, 227, 228, 229, 229, 230, 230, 231, 232, 232,
      232, 232, 233, 234, 235, 236, 236, 236, 236, 236, 236, 236, 236, 237, 238,
      239, 239, 239, 239, 239, 240, 241, 242, 243, 243, 243, 243, 243, 243, 244,
      244, 244, 244, 244, 244, 244, 244, 245, 246, 246, 247, 247, 248, 248, 248,
      249, 249, 249, 249, 249, 249, 249, 250, 251, 251, 252, 253, 253, 254, 255,
      255, 255, 256, 256, 257, 257, 258, 258, 258, 259, 260, 260, 261, 262, 262,
      263, 264, 265, 265, 265, 265, 266, 266, 267, 268, 268, 269, 269, 269, 270,
      270, 271, 272, 272, 273, 273, 274, 275, 276, 277, 277, 278, 279, 279, 279,
      279, 279, 279, 280, 281, 282, 283, 284, 284, 284, 285, 285, 286, 287, 287,
      287, 287, 288, 288, 288, 288, 288, 288, 288, 289, 289, 289, 290, 291, 291,
      291, 291, 292, 292, 292, 292, 292, 293, 293, 294, 295, 296, 297, 298, 298,
      299, 300, 301, 302, 302, 303, 303, 304, 304, 304, 304, 304, 304, 305, 306,
      307, 307, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 317, 317,
      317, 318, 318, 318, 318, 318, 319, 320, 321, 321, 321, 322, 323, 323, 323,
      323, 323, 324, 324, 324, 324, 324, 324, 325, 326, 327, 327, 327, 328, 329,
      329, 329, 329, 330, 331, 331, 331, 331, 332, 333, 333, 333, 334, 335, 335,
      335, 336, 337, 337, 337, 338, 339, 340, 340, 340, 341, 342, 343, 343, 344,
      344, 344, 345, 346, 347, 347, 348, 349, 349, 350, 350, 351, 351, 352, 353,
      354, 355, 355, 356, 357, 357, 358, 359, 359, 360, 360, 361, 362, 362, 363,
      364, 364, 365, 365, 366, 367, 367, 367, 368, 368, 368, 369, 369, 369, 369,
      370, 370, 371, 372, 373, 374, 374, 375, 375, 375, 376, 377, 378, 378, 378,
      378, 378, 378, 379, 380, 381, 381, 382, 382, 382, 382, 383, 383, 384, 385,
      385, 386, 386, 386, 387, 387, 387, 387, 387, 387, 387, 388, 388, 389, 389,
      389, 390, 391, 391, 392, 393, 394, 395, 396, 397, 398, 398, 399, 400, 400,
      400, 400, 400, 401, 402, 403, 403, 403, 404, 404, 405, 405, 406, 407, 407,
      407, 407, 408, 409, 410, 411, 412, 412, 412, 413, 414, 414, 415, 416, 417,
      417, 418, 418, 419, 419, 420, 420, 420, 421, 422, 422, 422, 423, 424, 425,
      426, 426, 426, 427, 428, 428, 429, 430, 430, 431, 431, 431, 431, 431, 432,
      433, 433, 434, 434, 434, 435, 435, 436, 436, 437, 438, 438, 438, 438, 439,
      439, 440, 441, 441, 442, 443, 444, 445, 446, 446, 446, 447, 447, 447, 447,
      448, 449, 449, 450, 450, 451, 451, 451, 452, 452, 453, 453, 453, 454, 454,
      454, 455, 455, 455, 456, 456, 456, 457, 457, 458, 458, 459, 460, 460, 461,
      461, 462, 462, 462, 463, 464, 464, 465, 466, 466, 467, 468, 468, 468, 469,
      470, 471, 471, 472, 472, 472, 472, 472, 473, 473, 474, 474, 475, 475, 475,
      475, 476, 476, 477, 477, 477, 477, 478, 479, 479, 479, 480, 480, 481, 481,
      482, 483, 483, 483, 484, 485, 485, 485, 486, 487, 488, 489, 489, 490, 490,
      491, 491, 491, 492, 493, 494, 495, 495, 495, 495, 496, 496, 496, 496, 496,
      496, 496, 496, 497, 498, 499, 499);

  VSET(10, e8, m1);
  VLOAD_8(v1, 0b00001110, 0b00000111, 0b00010000, 0b01001011, 0b00110100,
          0b01111101, 0b11001100, 0b00011000, 0b01000111, 0b00010100);
  VSET(77, e8, m1);
  asm volatile("viota.m v2, v1");
  VCMP_U8(6, v2, 0, 0, 1, 2, 3, 3, 3, 3, 3, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
          7, 7, 7, 7, 8, 9, 9, 10, 10, 10, 11, 11, 11, 11, 12, 12, 13, 14, 14,
          14, 15, 15, 16, 17, 18, 19, 20, 20, 20, 20, 21, 22, 22, 22, 23, 24,
          24, 24, 24, 25, 26, 26, 26, 26, 27, 28, 29, 29, 29, 29, 30, 30, 30,
          30, 31, 31);

  VSET(10, e8, m1);
  VLOAD_8(v0, 0b00001110, 0b00000111, 0b00010000, 0b01001011, 0b00110100,
          0b01111101, 0b11001100, 0b00011000, 0b01000111, 0b00010100);
  VSET(77, e32, m2);
  asm volatile("viota.m v2, v0");
  VCMP_U32(7, v2, 0, 0, 1, 2, 3, 3, 3, 3, 3, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
           7, 7, 7, 7, 8, 9, 9, 10, 10, 10, 11, 11, 11, 11, 12, 12, 13, 14, 14,
           14, 15, 15, 16, 17, 18, 19, 20, 20, 20, 20, 21, 22, 22, 22, 23, 24,
           24, 24, 24, 25, 26, 26, 26, 26, 27, 28, 29, 29, 29, 29, 30, 30, 30,
           30, 31, 31);

  VSET(10, e8, m1);
  VLOAD_8(v0, 0b00001110, 0b00000111, 0b00010000, 0b01001011, 0b00110100,
          0b01111101, 0b11001100, 0b00011000, 0b01000111, 0b00010100);
  VSET(77, e64, m4);
  asm volatile("viota.m v4, v0");
  VCMP_U64(8, v4, 0, 0, 1, 2, 3, 3, 3, 3, 3, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
           7, 7, 7, 7, 8, 9, 9, 10, 10, 10, 11, 11, 11, 11, 12, 12, 13, 14, 14,
           14, 15, 15, 16, 17, 18, 19, 20, 20, 20, 20, 21, 22, 22, 22, 23, 24,
           24, 24, 24, 25, 26, 26, 26, 26, 27, 28, 29, 29, 29, 29, 30, 30, 30,
           30, 31, 31);

  VSET(5, e16, m1);
  VLOAD_16(v0, 0b0000011100001110, 0b0100101100010000, 0b0111110100110100,
           0b0001100011001100, 0b0001010001000111);
  VSET(77, e8, m4);
  asm volatile("viota.m v4, v0");
  VCMP_U8(9, v4, 0, 0, 1, 2, 3, 3, 3, 3, 3, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
          7, 7, 7, 7, 8, 9, 9, 10, 10, 10, 11, 11, 11, 11, 12, 12, 13, 14, 14,
          14, 15, 15, 16, 17, 18, 19, 20, 20, 20, 20, 21, 22, 22, 22, 23, 24,
          24, 24, 24, 25, 26, 26, 26, 26, 27, 28, 29, 29, 29, 29, 30, 30, 30,
          30, 31, 31);
}

void TEST_CASE2() {
  VSET(16, e8, m1);
  VCLEAR(v2);
  VLOAD_8(v2, 0, 1, 2, 3, 4, 5, 6, 7);
  VSET(1, e8, m1);
  VLOAD_8(v1, 0b10001001);
  VLOAD_8(v0, 0b11000111);
  VSET(16, e8, m1);
  asm volatile("viota.m v2, v1, v0.t");
  VCMP_U8(10, v2, 0, 1, 1, 3, 4, 5, 1, 1);

  VSET(5, e16, m1);
  VLOAD_16(v8, 0b0000011100001110, 0b0100101100010000, 0b0111110100110100,
           0b0001100011001100, 0b0001010001000111);
  VSET(10, e16, m1);
  VLOAD_8(v0, 0b11111111, 0b11111111, 0b11111111, 0b11111111, 0b11111111,
          0b11111111, 0b11111111, 0b11111111, 0b11111111, 0b00001111);
  VSET(77, e8, m4);
  VCLEAR(v4);
  asm volatile("viota.m v4, v8, v0.t");
  VCMP_U8(11, v4, 0, 0, 1, 2, 3, 3, 3, 3, 3, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
          7, 7, 7, 7, 8, 9, 9, 10, 10, 10, 11, 11, 11, 11, 12, 12, 13, 14, 14,
          14, 15, 15, 16, 17, 18, 19, 20, 20, 20, 20, 21, 22, 22, 22, 23, 24,
          24, 24, 24, 25, 26, 26, 26, 26, 27, 28, 29, 29, 29, 29, 30, 30, 30,
          30, 31, 0);

  VSET(5, e16, m1);
  VLOAD_16(v8, 0b0000011100001110, 0b0100101100010000, 0b0111110100110100,
           0b0001100011001100, 0b0001010001000111);
  VSET(10, e16, m1);
  VLOAD_8(v0, 0b11111111, 0b11111111, 0b11111111, 0b11111111, 0b11111111,
          0b11111111, 0b11111111, 0b11111111, 0b11111111, 0b00001011);
  VSET(77, e8, m4);
  VCLEAR(v4);
  asm volatile("viota.m v4, v8, v0.t");
  VCMP_U8(12, v4, 0, 0, 1, 2, 3, 3, 3, 3, 3, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
          7, 7, 7, 7, 8, 9, 9, 10, 10, 10, 11, 11, 11, 11, 12, 12, 13, 14, 14,
          14, 15, 15, 16, 17, 18, 19, 20, 20, 20, 20, 21, 22, 22, 22, 23, 24,
          24, 24, 24, 25, 26, 26, 26, 26, 27, 28, 29, 29, 29, 29, 30, 30, 30, 0,
          30, 0);

  VSET(5, e16, m1);
  VLOAD_16(v8, 0b0000011100001110, 0b0100101100010000, 0b0111110100110100,
           0b0001100011001100, 0b0001010001000111);
  VSET(10, e16, m1);
  VLOAD_8(v0, 0b11111111, 0b11111111, 0b11111111, 0b11111111, 0b11111111,
          0b11111111, 0b11111111, 0b11111111, 0b11111111, 0b11111111);
  VSET(77, e8, m4);
  asm volatile("viota.m v4, v8, v0.t");
  VCMP_U8(13, v4, 0, 0, 1, 2, 3, 3, 3, 3, 3, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
          7, 7, 7, 7, 8, 9, 9, 10, 10, 10, 11, 11, 11, 11, 12, 12, 13, 14, 14,
          14, 15, 15, 16, 17, 18, 19, 20, 20, 20, 20, 21, 22, 22, 22, 23, 24,
          24, 24, 24, 25, 26, 26, 26, 26, 27, 28, 29, 29, 29, 29, 30, 30, 30,
          30, 31, 31);
}

int main(void) {
  INIT_CHECK();
  enable_vec();
  enable_fp();
  TEST_CASE1();
  TEST_CASE2();
  EXIT_CHECK();
}
